详解Numpy eye()（返回对角线上为1的数组）函数的作用与使用方法

2024-04-18 21:26| 来源: 网络整理| 查看: 265

Numpy eye()函数是用来创建一个二维数组，其主对角线上的元素都是1，其他元素都是0。

该函数的语法如下：

numpy.eye(N, M=None, k=0, dtype=, order='C')

其中，参数N表示二维数组的行数，M表示列数，默认为N；k表示主对角线偏移量，默认为0；dtype表示数据类型，可选值为float、int、complex等；order表示数组在内存中的存储顺序，可选值为'C'和'F'。

下面给出两个例子来说明Numpy eye()函数的用法：

例1：创建一个3*3的单位矩阵 import numpy as np arr = np.eye(3) print(arr)

输出结果为：

[[1. 0. 0.] [0. 1. 0.] [0. 0. 1.]]

可以看到，该代码创建了一个3*3的单位矩阵，主对角线上的元素都是1，其他元素都是0。

例2：创建一个3*4的右移2位的对角矩阵 import numpy as np arr = np.eye(3, 4, k=2) print(arr)

输出结果为：

[[0. 0. 1. 0.] [0. 0. 0. 1.] [0. 0. 0. 0.]]

可以看到，该代码创建了一个3*4的右移2位的对角矩阵，主对角线上的元素都是1，其他元素都是0。

本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：详解Numpy eye()（返回对角线上为1的数组）函数的作用与使用方法 - Python技术站

【本文地址】

今日新闻